package com.leetcode;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

/**
 * 202. 快乐数
 * 只有一个循环：4→16→37→58→89→145→42→20→4。所有其他数字都在进入这个循环的链上，或者在进入 1 的链上
 * 可以硬编码一个包含这些数字的散列集，如果我们达到其中一个数字，那么我们就知道在循环中
 *
 * @author fy
 * @date 2022/4/6 15:21
 */
public class Solution202_2 {
    private static Set<Integer> cycle = new HashSet<>(Arrays.asList(4, 16, 37, 58, 89, 145, 42, 20));

    public boolean isHappy(int n) {
        while (n != 1 && !cycle.contains(n)) {
            n = getNext(n);
        }
        return n == 1;
    }

    private int getNext(int n) {
        int sum = 0;
        while (n > 0) {
            int m = n % 10;
            sum += m * m;
            n = n / 10;
        }
        return sum;
    }

}
